package com.yx.coredata.utils;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Enumeration;

import org.apache.tools.zip.ZipEntry;
import org.apache.tools.zip.ZipFile;


public class UnCompressor {


    /**
     * 解压zip或者rar包的内容到指定的目录下，可以处理其文件夹下包含子文件夹的情况
     *
     * @param zipFile_     要解压的zip或者rar包文件
     * @param outFile 解压后存放的目录
     */
    public static synchronized void unzip(File zipFile_, File outFile)
            throws IOException {
        if (!outFile.exists()) {
            outFile.mkdirs();
        }
        ZipFile zipFile = new ZipFile(zipFile_);
        Enumeration en = zipFile.getEntries();
        ZipEntry zipEntry = null;
        while (en.hasMoreElements()) {
            zipEntry = (ZipEntry) en.nextElement();
            if (zipEntry.isDirectory()) {
                // mkdir directory
                String dirName = zipEntry.getName();
                //System.out.println("=dirName is:=" + dirName + "=end=");
                dirName = dirName.substring(0, dirName.length() - 1);
                File f = new File(outFile.getPath() + File.separator + dirName);
                f.mkdirs();
            } else {
                //unzip file
                String strFilePath = outFile.getPath() + File.separator
                        + zipEntry.getName();
                File f = new File(strFilePath);
                //the codes remedified by can_do on 2010-07-02 =begin=
                // /////begin/////
                //判断文件不存在的话，就创建该文件所在文件夹的目录
                if (!f.exists()) {
                    String[] arrFolderName = zipEntry.getName().split("/");
                    String strRealFolder = "";
                    for (int i = 0; i < (arrFolderName.length - 1); i++) {
                        strRealFolder += arrFolderName[i] + File.separator;
                    }
                    strRealFolder = outFile.getPath() + File.separator
                            + strRealFolder;
                    File tempDir = new File(strRealFolder);
                    //此处使用.mkdirs()方法，而不能用.mkdir()
                    tempDir.mkdirs();
                }
                //////end///
                // the codes remedified by can_do on 2010-07-02 =end=
                f.createNewFile();
                InputStream in = zipFile.getInputStream(zipEntry);
                FileOutputStream out = new FileOutputStream(f);
                try {
                    int c;
                    byte[] by = new byte[BUFFEREDSIZE];
                    while ((c = in.read(by)) != -1) {
                        out.write(by, 0, c);
                    }
                    out.flush();
                } catch (IOException e) {
                    throw e;
                } finally {
                    out.close();
                    in.close();
                }
            }
        }
    }

    private static final int BUFFEREDSIZE = 1024;

    public static void main(String[] args) {

        try {
            UnCompressor.unzip(new File("d:/tests.zip"), new File("D:/tests"));

            //UnCompressor.unzip("d:/struts.rar", "D:/struts");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}